import slicer
from slicer.ScriptedLoadableModule import *
from slicer.util import TESTING_DATA_URL


#
# RSNA2012ProstateDemo
#


class RSNA2012ProstateDemo(ScriptedLoadableModule):
    """Uses ScriptedLoadableModule base class, available at:
    https://github.com/Slicer/Slicer/blob/main/Base/Python/slicer/ScriptedLoadableModule.py
    """

    def __init__(self, parent):
        ScriptedLoadableModule.__init__(self, parent)
        parent.title = "RSNA2012ProstateDemo"  # TODO make this more human readable by adding spaces
        parent.categories = ["Testing.TestCases"]
        parent.dependencies = []
        parent.contributors = ["Steve Pieper (Isomics)"]  # replace with "Firstname Lastname (Org)"
        parent.helpText = """
    This module was developed as a self test to perform the operations needed for the RSNA 2012 Prostate Demo
    """
        parent.acknowledgementText = """
    This file was originally developed by Steve Pieper, Isomics, Inc.  and was partially funded by NIH grant 3P41RR013218-12S1.
"""  # replace with organization, grant and thanks.


#
# qRSNA2012ProstateDemoWidget
#


class RSNA2012ProstateDemoWidget(ScriptedLoadableModuleWidget):
    """Uses ScriptedLoadableModuleWidget base class, available at:
    https://github.com/Slicer/Slicer/blob/main/Base/Python/slicer/ScriptedLoadableModule.py
    """

    def setup(self):
        ScriptedLoadableModuleWidget.setup(self)
        # Instantiate and connect widgets ...

        # Add vertical spacer
        self.layout.addStretch(1)

    def cleanup(self):
        pass


class RSNA2012ProstateDemoTest(ScriptedLoadableModuleTest):
    """
    This is the test case for your scripted module.
    Uses ScriptedLoadableModuleTest base class, available at:
    https://github.com/Slicer/Slicer/blob/main/Base/Python/slicer/ScriptedLoadableModule.py
    """

    def setUp(self):
        slicer.mrmlScene.Clear(0)

    def runTest(self):
        self.setUp()
        self.test_RSNA2012ProstateDemo()

    def test_RSNA2012ProstateDemo(self):
        """Replicate one of the crashes in issue 2512"""

        print("Running RSNA2012ProstateDemo Test case:")

        import SampleData

        SampleData.downloadFromURL(
            fileNames="RSNA2012ProstateDemo.mrb",
            loadFiles=True,
            uris=TESTING_DATA_URL + "SHA256/2627388ee213564f8783d0242993212ba01189f4c6640d57c4cde4e28fc5f97b",
            checksums="SHA256:2627388ee213564f8783d0242993212ba01189f4c6640d57c4cde4e28fc5f97b")

        # get all scene view nodes and test switching
        svns = slicer.util.getNodes("vtkMRMLSceneViewNode*")

        for reps in range(5):
            for svname, svnode in svns.items():
                self.delayDisplay("Restoring scene view %s ..." % svname)
                svnode.RestoreScene()
                self.delayDisplay("OK")

        self.delayDisplay("Done testing scene views, will clear the scene")
        slicer.mrmlScene.Clear(0)
        self.delayDisplay("Test passed")
